Apply a function to every element in NumPy Array